package 哈希表系列;

import java.util.HashMap;
import java.util.Map;

public class FourSumCount {
    //暴力死循环，过不了
//    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
//        int sum = 0;
//        for(int i : nums1)
//            for(int j : nums2)
//                for(int k : nums3)
//                    for(int m : nums4)
//                        if(i+j+k+m == 0)
//                            sum++;
//        return sum;
//    }

    //HashMap
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int sum = 0;
        Map<Integer,Integer> map = new HashMap<>();
        //统计两个数组中的元素之和，同时统计出现的次数，放入map
        for (int i : nums1){
            for(int j : nums2){
                int temp = i + j;
                if(map.containsKey(temp)){  //如果已经存在temp
                    map.put(temp,map.get(temp)+1);     //则value值数量加1
                }else {
                    map.put(temp,1);    //如果不存在，则等于1
                }
            }
        }

        //统计剩余的两个元素的和，在map中找是否存在相加为0的情况，同时记录次数
        for(int i : nums3){
            for (int j : nums4){
                int temp = i + j;
                if(map.containsKey(0-temp)){
                    sum += map.get(0-temp);
                }
            }
        }

        return sum;
    }

}
